Python snippet showing weird behaviour